package org.example.system.service.impl;


import cn.hutool.core.collection.CollectionUtil;
import cn.hutool.core.util.ObjectUtil;
import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import lombok.RequiredArgsConstructor;
import org.example.system.common.constant.SystemConstants;
import org.example.system.common.model.Option;
import org.example.system.converter.RoleConverter;
import org.example.system.mapper.SysRoleMapper;
import org.example.system.model.entity.SysRole;
import org.example.system.model.entity.SysRoleMenu;
import org.example.system.model.form.RoleForm;
import org.example.system.model.query.RolePageQuery;
import org.example.system.model.vo.RolePageVO;
import org.example.system.security.util.SecurityUtils;
import org.example.system.service.SysRoleMenuService;
import org.example.system.service.SysRoleService;
import org.example.system.service.SysUserRoleService;
import org.springframework.cache.annotation.CacheEvict;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.function.Function;
import java.util.stream.Collectors;

/**
 * 角色业务实现类
 *
 * @author LY
 */
@Service
@RequiredArgsConstructor
public class SysRoleServiceImpl extends ServiceImpl<SysRoleMapper, SysRole> implements SysRoleService {

    private final SysRoleMenuService roleMenuService;
    private final SysUserRoleService userRoleService;
    private final RoleConverter roleConverter;

    /**
     * 角色分页列表
     *
     * @param queryParams 角色查询参数
     * @return {@link Page<RolePageVO>} – 角色分页列表
     */
    @Override
    public Page<RolePageVO> getRolePage(RolePageQuery queryParams) {
        // 查询参数
        int pageNum = queryParams.getPageNum();
        int pageSize = queryParams.getPageSize();
        String keywords = queryParams.getKeywords();

        // 查询数据
        Page<SysRole> rolePage = this.page(new Page<>(pageNum, pageSize),
                new LambdaQueryWrapper<SysRole>()
                        .and(StrUtil.isNotBlank(keywords),
                                wrapper ->
                                        wrapper.like(StrUtil.isNotBlank(keywords), SysRole::getName, keywords)
                                                .or()
                                                .like(StrUtil.isNotBlank(keywords), SysRole::getCode, keywords)
                        )
                        .ne(!SecurityUtils.isRoot(), SysRole::getCode, SystemConstants.ROOT_ROLE_CODE)
        );

        // 实体转换
        return roleConverter.entity2Page(rolePage);
    }

    /**
     * 角色下拉列表
     *
     * @return {@link List<Option>} – 角色下拉列表
     */
    @Override
    public List<Option> listRoleOptions() {
        // 查询数据
        List<SysRole> roleList = this.list(new LambdaQueryWrapper<SysRole>()
                .ne(!SecurityUtils.isRoot(), SysRole::getCode, SystemConstants.ROOT_ROLE_CODE)
                .select(SysRole::getId, SysRole::getName)
                .orderByAsc(SysRole::getSort)
        );

        // 实体转换
        return roleConverter.entities2Options(roleList);
    }

    /**
     * 保存角色
     *
     * @param roleForm 角色表单数据
     * @return 保存结果
     */
    @Override
    @Transactional(rollbackFor = Exception.class)
    public boolean saveRole(RoleForm roleForm) {

        Long roleId = roleForm.getId();

        // 编辑角色时，判断角色是否存在
        SysRole oldRole = null;
        if (roleId != null) {
            oldRole = this.getById(roleId);
            if (oldRole == null) {
                throw new IllegalArgumentException("角色不存在");
            }
        }

        String roleCode = roleForm.getCode();
        long count = this.count(new LambdaQueryWrapper<SysRole>()
                .ne(roleId != null, SysRole::getId, roleId)
                .and(wrapper -> wrapper
                        .eq(SysRole::getCode, roleCode)
                        .or()
                        .eq(SysRole::getName, roleForm.getName())
                ));
        if (count > 0) {
            throw new IllegalArgumentException("角色名称或角色编码已存在，请修改后重试！");
        }

        // 实体转换
        SysRole role = roleConverter.form2Entity(roleForm);

        boolean result = this.saveOrUpdate(role);
        if (result && oldRole != null) {
            // 判断角色编码或状态是否修改，修改了则刷新权限缓存
            try {
                if (!StrUtil.equals(oldRole.getCode(), roleCode) ||
                        !ObjectUtil.equals(oldRole.getStatus(), roleForm.getStatus())) {
                    roleMenuService.refreshRolePermsCache(oldRole.getCode(), roleCode);
                }
            } catch (Exception e) {
                log.warn("刷新角色权限缓存失败");
                throw new IllegalArgumentException("刷新角色权限缓存失败", e);
            }
        }
        return result;
    }

    /**
     * 获取角色表单数据
     *
     * @param roleId 角色ID
     * @return {@link RoleForm} – 角色表单数据
     */
    @Override
    public RoleForm getRoleForm(Long roleId) {
        SysRole entity = this.getById(roleId);
        return roleConverter.entity2Form(entity);
    }

    /**
     * 批量删除角色
     *
     * @param ids 角色ID，多个使用英文逗号(,)分割
     * @return {@link Boolean}
     */
    @Override
    @Transactional
    public boolean deleteRoles(String ids) {
        if (StrUtil.isBlank(ids)) {
            throw new IllegalArgumentException("删除的角色ID不能为空");
        }
        List<Long> roleIds = Arrays.stream(ids.split(","))
                .map(String::trim)
                .filter(s -> StrUtil.isNotBlank(s) && StrUtil.isNumeric(s))
                .map(Long::parseLong)
                .distinct()
                .toList();

        if (roleIds.isEmpty()) {
            throw new IllegalArgumentException("无效的角色ID");
        }

        // 批量查询角色
        Map<Long, SysRole> roleMap = this.listByIds(roleIds).stream()
                .collect(Collectors.toMap(SysRole::getId, Function.identity()));

        for (Long roleId : roleIds) {
            SysRole role = roleMap.get(roleId);
            if (role ==  null) {
                throw new IllegalArgumentException("角色不存在");
            }

            // 判断角色是否被用户关联
            boolean isRoleAssigned = userRoleService.hasAssignedUsers(roleId);
            if (isRoleAssigned) {
                throw new IllegalArgumentException("角色【" + role.getName() + "】已分配用户，请先解除关联后删除");
            }

            boolean deleteResult = this.removeById(roleId);
            if (deleteResult) {
                // 删除成功，刷新权限缓存
                roleMenuService.refreshRolePermsCache(role.getCode());
            }
        }

        return true;
    }

    /**
     * 获取角色的菜单ID集合
     *
     * @param roleId 角色ID
     * @return 菜单ID集合(包括按钮权限ID)
     */
    @Override
    public List<Long> getRoleMenuIds(Long roleId) {
        return roleMenuService.listMenuIdsByRoleId(roleId);
    }

    /**
     * 修改角色的资源权限
     *
     * @param roleId  角色ID
     * @param menuIds 菜单ID集合
     * @return 执行结果
     */
    @Override
    @CacheEvict(cacheNames = "menu", key = "'routes'")
    @Transactional
    public boolean assignMenusToRole(Long roleId, List<Long> menuIds) {
        SysRole role = this.getById(roleId);
        if (role == null) {
            throw new IllegalArgumentException("角色不存在");
        }

        // 执行数据库操作（删除+新增）
        updateRoleMenus(roleId, menuIds);

        // 刷新缓存（单独事务控制）
        refreshRolePermsCache(role.getCode());

        return true;
    }

    /**
     * 更新角色菜单关系
     */
    public void updateRoleMenus(Long roleId, List<Long> menuIds) {
        // 删除原有菜单权限
        roleMenuService.remove(
                new LambdaQueryWrapper<SysRoleMenu>()
                        .eq(SysRoleMenu::getRoleId, roleId)
        );

        // 新增新菜单权限
        if (CollectionUtil.isNotEmpty(menuIds)) {
            List<SysRoleMenu> roleMenus = menuIds
                    .stream()
                    .map(menuId -> new SysRoleMenu(roleId, menuId))
                    .toList();
            roleMenuService.saveBatch(roleMenus);
        }
    }

    /**
     * 刷新角色权限缓存（独立事务，防止影响主流程）
     */
    public void refreshRolePermsCache(String roleCode) {
        roleMenuService.refreshRolePermsCache(roleCode);
    }

    /**
     * 获取最大范围的数据权限
     *
     * @param roles 角色编码集合
     * @return {@link Integer} – 数据权限范围
     */
    @Override
    public Integer getMaximumDataScope(Set<String> roles) {
        return this.baseMapper.getMaximumDataScope(roles);
    }
}
